package com.jing.yao.component.mapper.injected;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.jing.yao.component.mapper.methods.YaoInsertBatch;
import com.jing.yao.component.mapper.methods.UpdateFieldsById;

import java.util.List;

/**
 * 自定义注入
 * @author Yao
 * @since 2020-07-09
 */
public class YaoSqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new YaoInsertBatch());
        methodList.add(new UpdateFieldsById());
        return methodList;
    }
}
